package com.webserver.http;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class HttpServletResponse {
    private Socket socket;
    private int statusCode = 200;
    private String statusReason= "OK";
    private File contentFile;
    private Map<String,String> headers = new HashMap<>();

    public HttpServletResponse(Socket socket){
        this.socket = socket;
    }

    public void response() throws IOException {
        //1.状态行
        sendStatusLine();
        //2.响应头
        sendHeaders();
        //3.响应正文
        sendContent();
    }

    private void sendStatusLine() throws IOException {
        println("HTTP/1.1"+" "+statusCode+" "+statusReason);
        System.out.println("发送响应");
        System.out.println("HTTP/1.1"+" "+statusCode+" "+statusReason);
    }
    private void sendHeaders() throws IOException {

        Set<Map.Entry<String,String>> entries = headers.entrySet();
        for (Map.Entry<String,String> h : entries){
            println(h.getKey() + ": " + h.getValue());
            System.out.println(h.getKey()+": " + h.getValue());
        }
        println("");
    }
    private void sendContent() throws IOException {
        try(
                FileInputStream fis = new FileInputStream(contentFile);
        ){
            OutputStream out = socket.getOutputStream();
            byte[] data = new byte[1024*10];
            int len;
            while ((len = fis.read(data)) != -1){
                out.write(data,0,len);
        }


        }
    }



    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        byte[] data = line.getBytes(StandardCharsets.ISO_8859_1);

        out.write(data);
        out.write(13);
        out.write(10);
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getContentFile() {
        return contentFile;
    }

    public void setContentFile(File contentFile) {
        this.contentFile = contentFile;

        try {
            String type = Files.probeContentType(contentFile.toPath());
            if(type != null){
                addHeader("Content-Type",type);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        addHeader("Content-Length",contentFile.length()+"");
    }

    public void addHeader(String key,String value){
        headers.put(key,value);
    }
}
